python 统一两个子图的colorbar显示范围 |
您所在的位置：网站首页 › matplotlib 画图颜色统一 › python 统一两个子图的colorbar显示范围 |
python 统一两个子图的colorbar显示范围
|
使用matplotlib绘图，改变colorbar, 统一colorbar的显示范围，目的是观看两个图片的差别，首先原始代码 import matplotlib.pyplot as plt import matplotlib m = 5 plt.figure(figsize=(15,15)) plt.subplot(121) plt.imshow(data['output'][m].squeeze(), cmap='jet') plt.colorbar(shrink=0.5) plt.title('output') plt.subplot(122) plt.imshow(data['label'][m].squeeze(), cmap='jet') plt.colorbar(shrink=0.5) plt.title('label') plt.show()得到的图片展示如下：
由上图看到colorbar数值范围不一致，图片差异明显，肉眼直观较难比较 设置一致的colorbar，便于比较 import matplotlib.pyplot as plt import matplotlib m = 5 plt.figure(figsize=(15,15)) norm = matplotlib.colors.Normalize(vmin=0,vmax=0.8) # 设置colorbar显示的最大最小值 plt.subplot(121) plt.imshow(data['output'][m].squeeze(), cmap='jet',norm=norm) plt.colorbar(shrink=0.5) plt.title('output') plt.subplot(122) plt.imshow(data['label'][m].squeeze(), cmap='jet',norm=norm) plt.colorbar(shrink=0.5) plt.title('label') plt.show()出来的图如下：
设vmin=0,vmax=3,出图如下
统一colorbar显示范围，更好比较 |
【本文地址】
今日新闻 |
推荐新闻 |